package TestExercises06;

//IO流


import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;
import java.util.Random;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/*
制造假数据：
    需求：制造假数据也是开发中的一个能力，在各个网上爬取数据，是其中一个方法。
 */
public class Test01 {
    public static void main(String[] args) throws IOException {

        //1、定义变量记录网址
        String famialNameNet = "https://hanyu.baidu.com/shici/detail?pid=0b2f26d4c0ddb3ee693fdb1137ee1b0d&from=kg0";
        String boyNameNet = "http://www.haoming8.cn/baobao/10881.html";
        String girlNameNet = "http://www.haoming8.cn/baobao/7641.html";

        //2、爬取数据，把网址上所有的数据拼接成一个字符串
        String famialNameStr = WebCrawler(famialNameNet);
        String boyNameStr = WebCrawler(boyNameNet);
        String girlNameStr = WebCrawler(girlNameNet);

        //3、通过正则表达式，把其中符合要求的数据获取出来
        ArrayList<String> famialyNameTempList = getData(famialNameStr, "(.{4})(，|。)", 1);
        ArrayList<String> boyNameTempList = getData(boyNameStr, "([\\u4E00-\\u9FA5]{2})(、|。)", 1);
        ArrayList<String> girlNameTempList = getData(girlNameStr, "(.. ){4}..", 0);

        //4、处理数据
        //famialNameTempList（姓氏）
        //处理方案:把每一个姓氏拆开并添加到一个新的集合当中
        ArrayList<String> famialyNameList = new ArrayList<>();
        for (String str : famialyNameTempList) {
            for (int i = 0; i < str.length(); i++) {
                char c = str.charAt(i);
                famialyNameList.add(c + "");
            }
        }

        //boyNameTempList（男生的姓名）
        //处理方案:去除其中的重复元素
        ArrayList<String> boyNameList = new ArrayList<>();
        for (String str : boyNameTempList) {
            if (!boyNameList.contains(str)) {
                boyNameList.add(str);
            }
        }

        //girlNameTempList（女生的姓名）
        //处理方案:把里面的每一个元素用空格进行切割，得到每一个女生的名字
        ArrayList<String> girlNameList = new ArrayList<>();
        for (String str : girlNameTempList) {
            String[] arr = str.split(" ");
            for (int i = 0; i < arr.length; i++) {
                girlNameList.add(arr[i]);
            }
        }

        //5、生成数据
        //姓名（唯一）-性别-年龄
        ArrayList<String> list = getInfos(famialyNameList, boyNameList, girlNameList, 70, 50);
        Collections.shuffle(list);

        //6、写出数据
        BufferedWriter bw = new BufferedWriter(new FileWriter("D:\\工作\\工作\\src\\names.txt"));
        for (String str : list) {
            bw.write(str);
            bw.newLine();
        }
        bw.close();
    }

    /*
     *作用：
     *   获取男生和女生的信息:张三-男-23
     *
     *形参：
     *   参数一:装着姓氏的集合
     *   参数二:装着男生名字的集合
     *   参数三:装着女生名字的集合
     *   参数四:男生的个数
     *   参数五:女生的个数
     */

    public static ArrayList<String> getInfos(ArrayList<String> famialyNameList, ArrayList<String> boyNameList, ArrayList<String> girlNameList, int boyCount, int girlCount) {
        //1、生成男生不重复的名字
        HashSet<String> boyhs = new HashSet<>();
        while (true) {
            if (boyhs.size() == boyCount) {
                break;
            }
            //随机
            Collections.shuffle(famialyNameList);
            Collections.shuffle(boyNameList);
            boyhs.add(famialyNameList.get(0) + boyNameList.get(0));
        }

        //2、生成女生不重复的名字
        HashSet<String> girlhs = new HashSet<>();
        while (true) {
            if (girlhs.size() == girlCount) {
                break;
            }
            //随机
            Collections.shuffle(famialyNameList);
            Collections.shuffle(girlNameList);
            girlhs.add(famialyNameList.get(0) + girlNameList.get(0));
        }

        //3、张三-男-23
        //生成男生信息
        ArrayList<String> list = new ArrayList<>();
        Random r = new Random();
        for (String boyName : boyhs) {
            int age = r.nextInt(10) + 18;//18岁~27岁
            list.add(boyName + "-男-" + age);
        }

        //生成女生信息
        for (String girlName : girlhs) {
            int age = r.nextInt(8) + 18;//18岁~25岁
            list.add(girlName + "-女-" + age);
        }
        return list;
    }


    /*
     * 作用:根据正则表达式获取字符串中的数据
     * 参数一:
     *   完整的字符串
     * 参数二:
     *   正则表达式
     * 参数三:
     *   ???
     *
     * 返回值:
     *   真正想要的数据
     */
    public static ArrayList<String> getData(String Str, String regex, int index) {
        //1、创建集合存放数据
        ArrayList<String> list = new ArrayList<>();
        //2、按照正则表达式的规则，去获取数据
        Pattern pattern = Pattern.compile(regex);
        //按照pattern的规则，到str当中获取数据
        Matcher matcher = pattern.matcher(Str);
        while (matcher.find()) {
            list.add(matcher.group(index));
        }
        return list;
    }

    /*
     * 作用:
     *  从网络中爬取数据，把数据拼接成字符串返回
     * 形参:
     *   网址
     * 返回值:
     *   爬取到的所有数据
     */
    public static String WebCrawler(String net) throws IOException {
        //1、定义StringBuilder拼接爬取到的数据
        StringBuilder sb = new StringBuilder();
        //2、创建一个URL对象
        URL url = new URL(net);
        //3、链接上这个网络
        //细节:保证网络是畅通的，而且这个网址是可以链接上的。
        URLConnection conn = url.openConnection();
        //4、读取数据
        InputStreamReader isr = new InputStreamReader(conn.getInputStream());
        int ch;
        while ((ch = isr.read()) != -1) {
            sb.append((char) ch);
        }
        //5、释放资源
        isr.close();
        //6、把读取到的数据返回
        return sb.toString();
    }
}
